from configparser import ConfigParser
import yaml


# 下面会一直用到configParser，所以直接继承
class ConfigHander(ConfigParser):

    def __init__(self, file, encoding=None):
        # 原本是 config = ConfigParse（）
        # 可以直接 init，调用父类init，相当于实例父类
        super().__init__(file)

        # config.read(f)
        self.read(file, encoding=encoding)


# yaml封装
def my_yaml(file_path):
    with open(file_path, encoding="utf-8") as f:
        data = yaml.load(f.read(), Loader=yaml.FullLoader())
    return data


# 写入yaml
def write_yaml(file_path, data, encoding="utf-8"):
    # 写入用W模式，因为a 模式写入 ，会造成重复
    with open(file_path, mode="w", encoding=encoding) as f:
        yaml.dump(data, stream=f, allow_unicode=True)  # 写入显示中文
